import sys
from time import sleep
import pygame
import random
from settings import Settings
from game_stats import GameStats
from scoreboard import Scoreboard
from play_button import Button
from gameover_button import GameOverButton
from ship import Ship
from bullet import Bullet
from alien import Alien

class AlienInvasion:
    def __init__(self):
        pygame.init()
        self.clock = pygame.time.Clock()
        self.settings = Settings()
        self.screen = pygame.display.set_mode((self.settings.screen_width,self.settings.screen_height))
        pygame.display.set_caption('Alien Invasion')
        self.stats = GameStats(self)
        self.sb = Scoreboard(self)
        self.ship = Ship(self)
        self.bullets = pygame.sprite.Group()
        self.aliens = pygame.sprite.Group()
        self.game_launch_status = False
        self.game_over_status = False
        self.play_button = Button(self, "Play")
        self.gameOver_button = GameOverButton(self, "Game Over")
    
    def run_game(self):
        while True:
            # 触发事件
            self._check_events()
            if self.game_launch_status:
                # 更新飞船位置
                self.ship.update()
                # 更新子弹位置 && 删除越界子弹 && 删除击中外星人的子弹
                self._update_bullets()
                # 更新外星人位置
                self._update_aliens()
            # 更新屏幕
            self._update_screen()
            # 60帧刷新屏幕
            self.clock.tick(60)
    
    def _check_events(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                sys.exit()
            elif event.type == pygame.KEYDOWN:
                self._check_keydown_events(event)
            elif event.type == pygame.KEYUP:
                self._check_keyup_events(event)
            elif event.type == pygame.MOUSEBUTTONDOWN:
                mouse_pos = pygame.mouse.get_pos()
                self._check_play_button(mouse_pos)

    def _check_play_button(self, mouse_pos):
        # 测试mouse_pos是否在按钮矩形范围内
        button_clicked = self.play_button.rect.collidepoint(mouse_pos)
        if button_clicked and not self.game_launch_status:
            self.game_launch_status = True
            self.settings.initialize_dynamic_settings()
            # 统计重置
            self.stats.reset_stats()
            # self.sb.prep_score()
            # self.sb.prep_level()
            # self.sb.prep_ships()
            # 清空子弹和外星人
            self.bullets.empty()
            self.aliens.empty()
            # 创建飞船
            self._create_fleet()
            # 飞船位置在屏幕中间底部位置
            self.ship.center_ship()
            # 鼠标不可见
            pygame.mouse.set_visible(False)

    def _check_keydown_events(self, event):
        if event.key == pygame.K_RIGHT:
            self.ship.moving_right = True
        elif event.key == pygame.K_LEFT:
            self.ship.moving_left = True
        elif event.key == pygame.K_q:
            sys.exit()
        elif event.key == pygame.K_SPACE:
            self._fire_bullet()

    def _check_keyup_events(self, event):
        if event.key == pygame.K_RIGHT:
            self.ship.moving_right = False
        elif event.key == pygame.K_LEFT:
            self.ship.moving_left = False

    def _fire_bullet(self):
        # 限制子弹数量
        if len(self.bullets) < self.settings.bullet_allowed:
            new_bullet = Bullet(self)
            self.bullets.add(new_bullet)

    def _update_screen(self):
        self.screen.fill(self.settings.bg_color)
        # 游戏结束（显示游戏结束文字和当前分数和最高记录分数&&禁用键盘）
        if self.game_over_status:
            self.gameOver_button.draw_button()
        else:
            # 屏幕上显现子弹
            for bullet in self.bullets.sprites():
                bullet.draw_bullet()
            # 屏幕上显现飞船
            self.ship.blitme()
            # 屏幕上显现外星人
            self.aliens.draw(self.screen)
            # 绘制Play按钮
            if not self.game_launch_status:
                self.play_button.draw_button()
        # 显示分数
        # self.sb.show_score()
        # 将所有的Surface绘制到screen
        pygame.display.flip()

    def _update_bullets(self):
        # 更新子弹位置
        self.bullets.update()
        # 删除越界子弹
        for bullet in self.bullets.copy():
            if bullet.rect.bottom <= 0:
                self.bullets.remove(bullet)
        # 检查是否击中外星人，如果是就删除对应的子弹和外星人
        self._check_bullet_alien_collisions()
    
    # 检查是否击中外星人，如果是就删除对应的子弹和外星人
    def _check_bullet_alien_collisions(self):
        # 找到bullets和aliens组中所有碰撞的sprites并自动销毁
        collsions = pygame.sprite.groupcollide(self.bullets, self.aliens, True, True)
        # if collsions:
        #     for aliens in collsions.values():
                # self.stats.score += self.settings.alien_points * len(aliens)
                # self.sb.prep_score()
                # self.sb.check_high_score()
        # 外星人组中没有外星人
        if not self.aliens:
            # 生产外星人
            self._create_fleet()
            # 提高速度
            self.settings.increase_speed()
        #     # 提高等级
        #     self.stats.level += 1
        #     self.sb.prep_level()

    def _create_fleet(self):
        alien = Alien(self)
        alien_width, alien_height = alien.rect.size
        # 屏幕顶部随机生成外星人
        left_margin = (0, 0 - alien_height)
        right_margin = (self.settings.screen_width - alien_width, 0 - alien_height)
        alien_rect_midtop = random.randint(left_margin[0], right_margin[0])
        self._create_alien(alien, alien_rect_midtop, 0)
        
    def _create_alien(self, new_alien, x_position, y_position):
        new_alien.x = x_position
        new_alien.rect.x = x_position
        new_alien.rect.y = y_position
        self.aliens.add(new_alien)

    # 向下移动 && 移动到屏幕底部删除外星人并重新生成外星人 && 外星人和飞船之间碰撞外星人删除以及游戏结束 &&  
    def _update_aliens(self):
        # 更新外星人位置
        self.aliens.update()
        # 检查飞船是否移动到屏幕底部
        self._check_fleet_edges()
        # 检测外星人和飞船之间的碰撞
        if pygame.sprite.spritecollideany(self.ship, self.aliens):
            self._ship_hit()

        # self._check_aliens_bottom()

    # 检查飞船是否移动到屏幕底部
    def _check_fleet_edges(self):
        for alien in self.aliens.copy():
            if alien.check_edges():
                # 删除外星人
                self.aliens.remove(alien)
                # 直接退出
                self._ship_hit()
            

    # 飞船和外星人撞击 && 外星人到达屏幕底部
    def _ship_hit(self):
        self.stats.chance = self.stats.chance - 1
        if self.stats.chance != 0:
            # self.sb.prep_ships()
            # 清空子弹和外星人
            self.bullets.empty()
            self.aliens.empty()
            # 是否需要创建外星人???
            self._create_fleet()
            # 飞船放在屏幕中间底部位置
            self.ship.center_ship()
            sleep(0.5)
        else: # 游戏结束
            self.game_over_status = True

        pygame.mouse.set_visible(True)

if __name__ == '__main__':
    ai = AlienInvasion()
    ai.run_game()